from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer

# 构建ChatBot并指定Adapter
Chinese_bot = ChatBot(
    'CARBOT',
    storage_adapter='chatterbot.storage.SQLStorageAdapter',
    logic_adapters=[
        {
            'import_path': 'chatterbot.logic.BestMatch',
            'default_response': '嗯嗯嗯'
        }
    ]
)
'''
Chinese_bot = ChatBot(
    'M',
    logic_adapters=[
        'chatterbot.logic.MathematicalEvaluation',
        'chatterbot.logic.TimeLogicAdapter'
    ]
)
'''
trainer = ListTrainer(Chinese_bot)
trainer.train([
    '我的车辆有问题吗?',
    '有问题[车辆没有汽油]',
    '在干嘛呢?',
    '吃饭',
    '咋这么晚才吃呢',
    '现在你的车辆上出现了[车辆轮胎气压]问题',
    '明天出去玩吗',
    '有事',
    '我想去韩国',
    '去年我去过了',
    '晚安',
    '晚晚安安',
    '我不能发现你的车辆有问题',
    '对不起，我听不清你说的话。 [车辆状态][顾客服务][新车上市][活动] 其中是什么内容？'
])
while True:
    print('CARBot: ',Chinese_bot.get_response(input('我:')))